package com.example.utils.util;

import java.io.*;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**
 * TODO  文件夹压缩
 *
 * @Author 江
 * @Date 2021/9/26 下午 15:22
 */
public class FileZipUtil {

    public static void main(String[] args) {
        String source = "D:\\5GC";
        String sourceFilePath = source;
        System.out.println(sourceFilePath);
        String zipFilePath = "D:\\md2\\5GC-SIM.zip";
        long l = System.currentTimeMillis();
        System.out.println(l);
        try {
            unzzip(zipFilePath, sourceFilePath);
        } catch (Exception e) {
            e.printStackTrace();
        }
        long ll = System.currentTimeMillis();
        System.out.println(ll);
        System.out.println(ll - l);

    }

    /**
     *  TODO   文件夹压缩的工具类
     * @Param : zipFileName  压缩后的路径和名字
     * @Param : sourceFileName 需要压缩的文件夹的路径
     * @return String
     * @Author 江
     * @Date 2021/9/30 下午 12:08
     */
    public static String unzzip(String zipFileName, String sourceFileName) throws Exception {
        //File zipFile = new File(zipFileName);
        System.out.println("压缩中...");

        //创建zip输出流
        ZipOutputStream out = new ZipOutputStream(new FileOutputStream(zipFileName));

        //创建缓冲输出流
        BufferedOutputStream bos = new BufferedOutputStream(out);

        File sourceFile = new File(sourceFileName);

        //调用函数
        String compress = compress(out, bos, sourceFile, sourceFile.getName());
        if (!"ok".equals(compress)) {
            bos.close();
            out.close();
            System.out.println(compress);
            return compress;
        }

        bos.close();
        out.close();
        System.out.println("压缩完成");
        return "压缩完成";

    }

    public static String compress(ZipOutputStream out,
                                  BufferedOutputStream bos,
                                  File sourceFile, String base) throws Exception {
        //如果路径为目录（文件夹）
        if (sourceFile.isDirectory()) {

            //取出文件夹中的文件（或子文件夹）
            File[] flist = sourceFile.listFiles();

            //如果文件夹为空，则只需在目的地zip文件中写入一个目录进入点
            if (flist.length == 0){
                System.out.println(base + "/");
                out.putNextEntry(new ZipEntry(base + "/"));
            } else{  //如果文件夹不为空，则递归调用compress，文件夹中的每一个文件（或文件夹）进行压缩
                for (int i = 0; i < flist.length; i++) {
                    compress(out, bos, flist[i], base + "/" + flist[i].getName());
                }
            }
        } else if (sourceFile.exists()){ //如果不是目录（文件夹），即为文件，则先写入目录进入点，之后将文件写入zip文件中
            out.putNextEntry(new ZipEntry(base));
            FileInputStream fos = new FileInputStream(sourceFile);
            BufferedInputStream bis = new BufferedInputStream(fos);

            int tag;
            System.out.println(base);
            //将源文件写入到zip文件中
            while ((tag = bis.read()) != -1) {
                out.write(tag);
            }
            bis.close();
            fos.close();

        } else {
            return "文件路径不正确，请重新确认";
        }
        return "ok";
    }



}
